'''
Author: 奔跑的乌龟
Date: 2020-12-05 13:25:53
LastEditTime: 2020-12-05 14:44:44
email: 435327238@qq.com
FilePath: \waibaoe:\爬虫12-4\Db.py
'''
import pymysql

# 数据库操作类

def config():
    # 打开数据库连接
    db = pymysql.connect("localhost", "root", "root", "tushuwang")
    return db

# 插入数据
def insert(sql,value):
    
    # 引入data数据类
    db = config()

    # 使用 cursor() 方法创建一个游标对象 cursor
    cursor = db.cursor()

    # 执行sql语句
    cursor.execute(sql,value)

    # 提交到数据库执行
    db.commit()

    # 如果发生错误则回滚
    # db.rollback()

    # 关闭数据库连接
    db.close()

# 删除数据
def delete(value):
    # 引入data数据类
    db = config()

    # 使用 cursor() 方法创建一个游标对象 cursor
    cursor = db.cursor()

    # 执行sql语句
    cursor.execute(value)

    # 提交到数据库执行
    db.commit()

    # 如果发生错误则回滚
    # db.rollback()

    # 关闭数据库连接
    db.close()    

# 查询数据
def query(value):
    # 引入data数据类
    db = config()
    
    #  使用 cursor() 方法创建一个游标对象 cursor
    cursor = db.cursor()

    #使用 execute()  方法执行 SQL 查询,注意有引号
    cursor.execute(value)

    # 执行sql语句
    data = cursor.fetchall()

    return data
    # 提交到数据库执行
    db.commit()

    # 关闭数据库连接
    db.close()